package woa.recursion;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 杨辉三角 II
 * 给定一个非负索引 rowIndex，返回「杨辉三角」的第 rowIndex 行。
 * 在「杨辉三角」中，每个数是它左上方和右上方的数的和。
 *
 * 输入: rowIndex = 3
 * 输出: [1,3,3,1]
 *
 * @author wangpeng
 * @date 2021/12/14
 */
public class YhsjGetRow {

    /**
     * 从后往前遍历
     *
     * @param rowIndex
     * @return
     */
    public static List<Integer> getRow(int rowIndex) {
        List<Integer> rs = new ArrayList<>(rowIndex);
        for (int i = 0; i <= rowIndex; i++) {
            for (int j = i; j >= 0; j--) {
                if (i == j) {
                    rs.add(1);
                } else if (j > 0) {
                    rs.set(j, rs.get(j) + rs.get(j - 1));
                }
            }
        }
        return rs;
    }

    /**
     * 递归
     *
     * @param rowIndex
     * @return
     */
    public static List<Integer> getRow1(int rowIndex) {
        if (rowIndex == 0) {
            return new ArrayList<>(Arrays.asList(1));
        }
        if (rowIndex == 1) {
            return new ArrayList<>(Arrays.asList(1, 1));
        }
        List<Integer> rs = getRow1(rowIndex - 1);
        rs.add(1);
        for (int i = rs.size() - 2; i > 0; i--) {
            rs.set(i, rs.get(i) + rs.get(i - 1));
        }

        return rs;
    }

    public static void main(String[] args) {
        System.out.println(getRow1(4));
    }
}
